package com.atguigu.tingshu.user.service.impl;

import com.alibaba.fastjson.JSON;
import com.atguigu.tingshu.common.constant.SystemConstant;
import com.atguigu.tingshu.common.rabbit.constant.MqConst;
import com.atguigu.tingshu.common.rabbit.service.RabbitService;
import com.atguigu.tingshu.common.util.MongoUtil;
import com.atguigu.tingshu.model.user.UserListenProcess;
import com.atguigu.tingshu.user.service.UserListenProcessService;
import com.atguigu.tingshu.vo.album.TrackStatMqVo;
import com.atguigu.tingshu.vo.album.TrackStatVo;
import com.atguigu.tingshu.vo.user.UserListenProcessVo;
import com.mongodb.client.MongoClient;
import org.bson.types.ObjectId;
import org.springframework.beans.BeanUtils;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.data.domain.Sort;
import org.springframework.data.mongodb.core.MongoTemplate;
import org.springframework.data.mongodb.core.query.Criteria;
import org.springframework.data.mongodb.core.query.Query;
import org.springframework.data.redis.core.RedisTemplate;
import org.springframework.stereotype.Service;

import java.math.BigDecimal;
import java.util.Date;
import java.util.HashMap;
import java.util.Map;
import java.util.UUID;
import java.util.concurrent.TimeUnit;

@Service
@SuppressWarnings({"all"})
public class UserListenProcessServiceImpl implements UserListenProcessService {

	@Autowired
	private MongoTemplate mongoTemplate;
    @Autowired
    private MongoClient mongo;
    @Autowired
    private RedisTemplate redisTemplate;
    @Autowired
    private RabbitService rabbitService;

    @Override
    public BigDecimal getTrackBreakSecond(Long userId, Long trackId) {
//        - 思路：根据用户id、声音id从mongoDB中获取数据；
//        给集合起一个名字：userListenProcess_userId，存储数据时，将数据封装到
//        userListenProcess这个对象中。
//          - mongoDB提供的查询API叫Query。

//        1.组成集合名称：
            String collectionName = MongoUtil.getCollectionName(MongoUtil.MongoCollectionEnum.USER_LISTEN_PROCESS,userId);
//        2.从集合中获取对应的播放时长，userId和trackId
            Query query = Query.query(Criteria.where("trackId").is(trackId));
        UserListenProcess userListenProcess = mongoTemplate.findOne(query, UserListenProcess.class, collectionName);
//        3.判断是否查询到播放时长
        if (null != userListenProcess){
            return userListenProcess.getBreakSecond();
        }
        return new BigDecimal("0");
    }

    @Override
    public void updateListenProcess(Long userId, UserListenProcessVo userListenProcessVo) {
//        先查询MongoDB中是否有该声音的播放进度：
        String collectionName = MongoUtil.getCollectionName(MongoUtil.MongoCollectionEnum.USER_LISTEN_PROCESS, userId);
//        从mongodb的集合中获取对应的播放时长，userID and trackID：
        Query query = Query.query(Criteria.where("trackId").is(userListenProcessVo.getTrackId()));
        UserListenProcess userListenProcess = mongoTemplate.findOne(query, UserListenProcess.class, collectionName);
//        判断：
        if (null != userListenProcess){
//            覆盖写声音播放进度：
            userListenProcess.setBreakSecond(userListenProcessVo.getBreakSecond());
            userListenProcess.setUpdateTime(new Date());
    }else {
//            新增播放进度：
            userListenProcess = new UserListenProcess();
            userListenProcess.setUserId(userId);
            userListenProcess.setTrackId(userListenProcessVo.getTrackId());
            userListenProcess.setAlbumId(userListenProcessVo.getAlbumId());
            userListenProcess.setBreakSecond(userListenProcessVo.getBreakSecond());
            userListenProcess.setCreateTime(new Date());
            userListenProcess.setUpdateTime(new Date());
        }
//        保存数据：
        this.mongoTemplate.save(userListenProcess,collectionName);
//        利用分布式锁思想解决声音重复播放问题，避免恶意刷单，key=userid：trackid；value=24h
        String key = userId + ":" +userListenProcess.getTrackId();
        Boolean result = this.redisTemplate.opsForValue().setIfAbsent(key, "1", 24, TimeUnit.HOURS);
        if (result){
//            如果分布式锁key存在，说明24小时内这个用户在这个声音下是第一次播放，专辑和声音播放量+1
            TrackStatMqVo trackStatMqVo = new TrackStatMqVo();
            trackStatMqVo.setCount(1);
            trackStatMqVo.setStatType(SystemConstant.TRACK_STAT_PLAY);
            trackStatMqVo.setTrackId(userListenProcess.getTrackId());
            trackStatMqVo.setAlbumId(userListenProcess.getAlbumId());
            String uuid = UUID.randomUUID().toString().replaceAll("-","");
            trackStatMqVo.setBusinessNo(uuid);
            rabbitService.sendMessage(MqConst.EXCHANGE_ALBUM,MqConst.ROUTING_TRACK_STAT_UPDATE,JSON.toJSONString(trackStatMqVo));
        }
    }

    @Override
    public Map<String, Object> getLatelyTrack(Long userId) {
        HashMap<String, Object> map = new HashMap<>();
        String collectionName = MongoUtil.getCollectionName(MongoUtil.MongoCollectionEnum.USER_LISTEN_PROCESS,userId);
        Query query = new Query();
        query.with(Sort.by(Sort.Direction.DESC,"updateTime")).limit(1);
        UserListenProcess userListenProcess = mongoTemplate.findOne(query, UserListenProcess.class, collectionName);
//        判断：
        if (null != userListenProcess){
            map.put("albumId",userListenProcess.getAlbumId());
            map.put("trackId",userListenProcess.getTrackId());
        }
    return map;
    }
}
